identity function of numpy in python